import pygame

pygame.init()
screen = pygame.display.set_mode((400, 300))
pygame.display.set_caption("my game")
bg = (255, 255, 255)
clock = pygame.time.Clock()
done = False
is_blue = True
x = 30
y = 30


def set_movement():
    global x, y
    pressed = pygame.key.get_pressed()
    if pressed[pygame.K_UP]:
        y -= 3
    elif pressed[pygame.K_DOWN]:
        y += 3
    elif pressed[pygame.K_LEFT]:
        x -= 3
    elif pressed[pygame.K_RIGHT]:
        x += 3


while not done:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            done = True
        if event.type == pygame.KEYDOWN and event.key == pygame.K_SPACE:
            is_blue = not is_blue

    screen.fill(bg)
    set_movement()
    color = (0, 128, 255) if is_blue else (255, 100, 0)
    pygame.draw.rect(screen, color, pygame.Rect(x, y, 60, 60))
    pygame.display.flip()
    clock.tick(60)